import requests
import app


# 登录接口
class LoginApi:
    def __init__(self):
        self.url = app.BASE_URL + '/api/sys/login'
        self.headers = app.HEADERS

    def login(self, json_data):
        return requests.post(url=self.url, headers=self.headers, json=json_data)


# 员工管理接口
class EmpManageApi:
    def __init__(self):
        self.url = app.BASE_URL + '/api/sys/user'

    # 添加员工
    def add_emp(self, header, json_data):
        # 发送添加员工请求
        resp = requests.post(url=self.url, headers=header, json=json_data)

        # 获取员工id
        emp_id = resp.json().get('data').get('id')
        # 写入全局变量 --- app.py 中
        app.EMP_ID = emp_id

        return resp

    # 查询员工
    def query_emp(self, header):
        # 发送 查询员工请求， 使用 全局 EMP_ID
        resp = requests.get(url=self.url + '/' + app.EMP_ID, headers=header)
        # 返回响应结果
        return resp

    # 修改员工
    def modify_emp(self, header, json_data):
        # 发送 修改员工请求， 使用 全局 EMP_ID
        resp = requests.put(url=self.url + '/' + app.EMP_ID, headers=header, json=json_data)
        # 返回响应结果
        return resp

    # 删除员工
    def delete_emp(self, header):
        # 发送 删除员工请求， 使用 全局 EMP_ID
        resp = requests.delete(url=self.url + '/' + app.EMP_ID, headers=header)
        # 返回响应结果
        return resp

    # 查询员工列表
    def query_emp_list(self, header):
        # 发送 查询员工列表请求，
        resp = requests.get(url=self.url + '?page=1&size=10', headers=header)
        # 返回响应结果
        return resp
